import pygame
from settings import *


class Weapon(pygame.sprite.Sprite):
    def __init__(self, player, 精灵组):
        super().__init__(精灵组)
        self.player = player
        self.攻击类型 = '武器'
        self.方向 = self.player.动作.split('_')[0]

        武器路径 = f'../graphics/weapons/{self.player.武器}/{self.方向}.png'
        self.image = pygame.image.load(武器路径).convert_alpha()

        # 水平方向使用武器的时候需要让武器显示区域对比角色的区域要向下移动16个像素
        # 垂直方向使用武器的时候需要让武器显示区域对比角色的区域要向上移动10个像素
        if self.方向 == 'right':
            self.rect = self.image.get_rect(midleft=player.rect.midright + pygame.math.Vector2(0, 16))
        elif self.方向 == 'left':
            self.rect = self.image.get_rect(midright=player.rect.midleft + pygame.math.Vector2(0, 16))
        elif self.方向 == 'up':
            self.rect = self.image.get_rect(midbottom=player.rect.midtop + pygame.math.Vector2(-10, 0))
        else:
            self.rect = self.image.get_rect(midtop=player.rect.midbottom + pygame.math.Vector2(-10, 0))
